package com.huangyi;

public class Main {
    public static void main(String[] args) {
        //除自身以外数组的乘积
        class Solution {
            public int[] productExceptSelf(int[] nums) {
                int n = nums.length;

                // 前缀积 & 后缀积数组
                int[] lprod = new int[n];
                int[] rprod = new int[n];

                // 初始化边界
                lprod[0] = 1;
                rprod[n - 1] = 1;

                // 计算前缀积
                for (int i = 1; i < n; i++) {
                    lprod[i] = lprod[i - 1] * nums[i - 1];
                }

                // 计算后缀积
                for (int i = n - 2; i >= 0; i--) {
                    rprod[i] = rprod[i + 1] * nums[i + 1];
                }

                // 计算最终结果
                int[] answer = new int[n];
                for (int i = 0; i < n; i++) {
                    answer[i] = lprod[i] * rprod[i];
                }

                return answer;
            }
        }


        //寻找数组的中心下标
        class Solution2 {
            public int pivotIndex(int[] nums) {
                int n = nums.length;

                // lsum[i] 表示 [0, i-1] 区间和
                // rsum[i] 表示 [i+1, n-1] 区间和
                int[] lsum = new int[n];
                int[] rsum = new int[n];

                // 构造前缀和数组
                for (int i = 1; i < n; i++) {
                    lsum[i] = lsum[i - 1] + nums[i - 1];
                }

                // 构造后缀和数组
                for (int i = n - 2; i >= 0; i--) {
                    rsum[i] = rsum[i + 1] + nums[i + 1];
                }

                // 判断中心下标
                for (int i = 0; i < n; i++) {
                    if (lsum[i] == rsum[i]) {
                        return i;
                    }
                }

                return -1;
            }
        }
    }
}